package p01_sparseArray;

import Utils.io.IOUtils;
import org.junit.jupiter.api.Test;
import utils.myUtils;

import java.io.IOException;

public class P1 {
    @Test
    public void test01() throws IOException {
        //创達一个原始的二维数组11 * 11
        // 0:表示没有棋子，1表示黑子 2表蓝子
        int[][] chessArr = new int[11][11];
        chessArr[1][2] = 1;
        chessArr[2][3] = 2;
        myUtils.print2dArray(chessArr);

        //将二维数组转稀晓数组的思
        //1.先遍历二维数组得到非0数据的个数
        int sum = 0;
        for (int[] row : chessArr) {
            for (int cell : row) {
                if (cell != 0) {
                    sum++;
                }
            }
        }

        // 2.创建对应的稀疏数组
        int[][] sparseArr = new int[sum + 1][3];
        sparseArr[0][0] = 11;
        sparseArr[0][1] = 11;
        sparseArr[0][2] = sum;
        int r = 1;
        for (int i = 0; i < chessArr.length; i++) {
            for (int j = 0; j < chessArr[i].length; j++) {
                if (chessArr[i][j] != 0) {
                    sparseArr[r][0] = i;
                    sparseArr[r][1] = j;
                    sparseArr[r][2] = chessArr[i][j];
                    r++;
                }
            }
        }
        System.out.println("================");
        myUtils.print2dArray(sparseArr);

        //将稀疏数组恢复成原始的二维数组
        int[][] chessArr1 = new int[sparseArr[0][0]][sparseArr[0][1]];
        for (int i = 1; i < sparseArr.length; i++) {
            chessArr1[sparseArr[i][0]][sparseArr[i][1]] = sparseArr[i][2];
        }
        System.out.println("================");
        myUtils.print2dArray(chessArr1);

        // 将稀疏数组写入文件
        String path = "D:\\IdeaProjects\\Practice2022-2-3\\DataStructures\\src\\test\\resources\\sparseArray\\map.data";
        IOUtils.String2File(myUtils.twoDArray2String(sparseArr), path);

        // 从文件读入稀疏数组
        String data = IOUtils.file2String(path);
        String[] rows = data.split("\n");
        String[] row1st = rows[0].split("\t");
        int[][] arrRes = new int[Integer.parseInt(row1st[0])][Integer.parseInt(row1st[1])];
        for (int i = 1; i < rows.length; i++) {
            String[] cells = rows[i].split("\t");
            arrRes[Integer.parseInt(cells[0])][Integer.parseInt(cells[1])] = Integer.parseInt(cells[2]);
        }
        System.out.println("从文件读入稀疏数组");
        myUtils.print2dArray(arrRes);
    }
}
